package com.fy.proj2_login_api.controller;

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Base64;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import com.fy.proj2_login_api.util.CaptchaUtil;

// 验证码控制器类，提供验证码生成和验证功能
@RestController
@RequestMapping("/api/captcha")
public class CaptchaController {
    
    // 使用ConcurrentHashMap存储验证码，支持高并发访问
    private final ConcurrentHashMap<String, CaptchaInfo> captchaStore = new ConcurrentHashMap<>();
    
    // 使用定时执行器服务来清理过期的验证码
    private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
    
    // 验证码信息内部类，存储验证码文本和创建时间
    private static class CaptchaInfo {
        // 验证码文本
        private final String text;
        // 创建时间戳
        private final long createTime;
        
        // 构造函数，初始化验证码文本和创建时间
        public CaptchaInfo(String text) {
            this.text = text;
            this.createTime = System.currentTimeMillis();
        }
        
        // 获取验证码文本
        public String getText() {
            return text;
        }
        
        // 获取创建时间
        public long getCreateTime() {
            return createTime;
        }
    }
    
    // 构造函数，初始化定时清理任务
    public CaptchaController() {
        // 每分钟执行一次清理任务
        scheduler.scheduleAtFixedRate(this::cleanExpiredCaptchas, 1, 1, TimeUnit.MINUTES);
    }
    
    // 清理过期的验证码
    private void cleanExpiredCaptchas() {
        // 获取当前时间
        long currentTime = System.currentTimeMillis();
        // 设置验证码过期时间为5分钟
        long expireTime = 5 * 60 * 1000; // 5分钟过期
        
        // 遍历验证码存储
        captchaStore.forEach((id, info) -> {
            // 检查验证码是否过期
            if (currentTime - info.getCreateTime() > expireTime) {
                // 移除过期的验证码
                captchaStore.remove(id);
            }
        });
    }

    // 生成验证码接口
    @GetMapping(value = "/generate", produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<?> generateCaptcha() {
        try {

            // 生成验证码文本
            String captchaText = CaptchaUtil.generateCaptchaCode();
            // 生成验证码图片
            BufferedImage image = CaptchaUtil.generateCaptchaImage(captchaText);
            
            // 将图片转换为字节数组输出流
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            // 写入PNG格式的图片
            javax.imageio.ImageIO.write(image, "png", baos);
            // 获取图片字节数组
            byte[] imageBytes = baos.toByteArray();
            // 将字节数组转换为Base64编码
            String base64Image = Base64.getEncoder().encodeToString(imageBytes);
            
            // 生成唯一的验证码ID
            String captchaId = UUID.randomUUID().toString();
            // 将验证码信息存入存储
            captchaStore.put(captchaId, new CaptchaInfo(captchaText));
            
            // 创建响应Map
            Map<String, Object> response = new HashMap<>();
            // 添加验证码ID
            response.put("captchaId", captchaId);
            // 添加Base64编码的图片
            response.put("image", "data:image/png;base64," + base64Image);
            
            // 返回成功响应
            return ResponseEntity.ok(response);
        } catch (IOException e) {
            // 发生IO异常，返回500状态码
            return ResponseEntity.status(500).body("生成验证码失败");
        }
    }
    
    // 验证验证码接口
    @PostMapping("/validate")
    public ResponseEntity<?> validateCaptcha(@RequestBody Map<String, String> request) {
        // 从请求体中获取验证码ID
        String captchaId = request.get("captchaId");
        // 从请求体中获取用户输入的验证码
        String captchaInput = request.get("captchaInput");
        
        try {
            // 检查验证码ID是否存在
            if (!captchaStore.containsKey(captchaId)) {
                // 验证码ID不存在，返回400状态码
                return ResponseEntity.status(400).body("验证码ID不存在或已过期");
            }
            
            // 获取验证码信息
            CaptchaInfo captchaInfo = captchaStore.get(captchaId);
            // 获取存储的验证码文本
            String storedCaptcha = captchaInfo.getText();
            
            // 验证用户输入是否正确（不区分大小写）
            boolean isValid = storedCaptcha.equalsIgnoreCase(captchaInput);
            
            // 创建响应Map
            Map<String, Object> response = new HashMap<>();
            // 添加验证结果
            response.put("valid", isValid);
            
            // 如果验证成功，从存储中删除验证码（一次性使用）
            if (isValid) {
                captchaStore.remove(captchaId);
            }
            
            // 返回验证结果
            return ResponseEntity.ok(response);
        } catch (Exception e) {
            // 发生异常，返回500状态码
            return ResponseEntity.status(500).body("验证码验证失败");
        }
    }
}